# 10、学员管理系统—函数应用
# （1）系统简介
# 需求：进入系统显示系统功能界面，功能如下：
# 1、添加学员
# 2、删除学员
# 3、修改学员信息
# 4、查询学员信息
# 5、显示所有学员信息
# 6、退出系统
# 系统共6个功能，用户根据自己需求选取


def welcome():
    print('-' * 20)
    print('欢迎登录学员管理系统')
    print('1：添加学员')
    print('2：删除学员')
    print('3：修改学员信息')
    print('4：查询学员信息')
    print('5：显示所有学员信息')
    print('6：退出系统')
    print('-' * 20)


def add_info():
    """
    添加学员
    :return:
    """
    new_id = input('学号：')
    new_name = input('姓名:')
    new_tel = input('手机:')
    global info
    for i in info:
        if new_name == i['name']:
            print('用户已存在')
            return
    else:
        info_dict = {'id': new_id, 'name': new_name, 'tel': new_tel}
        info.append(info_dict)
        print(info)


def del_info():
    """
    删除学员
    :return:
    """
    del_name = input('请输入待删除的学员姓名：')
    global info
    for i in info:
        if del_name == i['name']:
            info.remove(i)
            break
        else:
            print('学员不存在')
        print(info)


def modify_info():
    """
    修改学员
    :return:
    """
    modify_name = input('请输入待修改的学员姓名：')
    global info
    for i in info:
        if modify_name == i['name']:
            i['tel'] = input('请输入新的手机号：')
            print(f"学号：{i['id']},姓名：{i['name']},手机{'tel'}")
            break
        else:
            print('学员不存在')
        print(info)


def select_info():
    """
    查询学员
    :return: 
    """""
    select_name = input('请输入待查找的学生姓名：')
    global info
    for i in info:
        if select_name == i['name']:
            print(f"学号：{i['id']},姓名：{i['name']},手机{'tel'}")
            break
        else:
            print('学员不存在')


def select_all():
    """
    显示所有学员
    :return:
    """
    print('学号\t姓名\t手机号')
    for i in info:
        print(f"{i['id']}\t{i['name']}\t{i['tel']}")


info = []
while True:
    welcome()
    user_num = input('选择要输入的功能序号:')
    if user_num == '1':
        print('添加学员')
        add_info()
    elif user_num == '2':
        print('删除学员')
        del_info()
    elif user_num == '3':
        print('修改学员信息')
        modify_info()
    elif user_num == '4':
        print('查询学员信息')
        select_info()
    elif user_num == '5':
        print('显示所有学员信息')
        select_all()
    elif user_num == '6':
        flag = input('确定要退出吗，yes为退出')
        if flag == 'yes':
            break
    else:
        print('输入格式错误,请重新输入')
